package org.richin.dao.util;
import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.util.List;

import org.richin.file.util.FileUtils;
import org.richin.io.util.IOUtils;
/**
 * 备份和导入是一个互逆的过程。 
 * 备份：程序调用mysql的备份命令，读出控制台输入流信息，写入.sql文件；
 * 导入：程序调用mysql的导入命令，把从.sql文件中读出的信息写入控制台的输出流 注意：此时定向符">"和"<"是不能用的
 * 前提条件，设置mysql的环境变量（在path中添加%MYSQL_HOME%\bin）
 * @author Administrator
 *
 */
public class MysqlDBUtil {
	/**
	 * 备份数据库,生成SQL文件
	 */
	public static void backup(String username,String password,String database,String charset,String filepath) {
		try {
			Runtime rt = Runtime.getRuntime();
			// 调用 mysql 的 cmd:
			String cmd = "mysqldump -u"+username+" -p"+password+" --set-charset=utf8 "+database;
			Process child = rt.exec(cmd);// 设置导出编码为utf8。这里必须是utf8

			// 把进程执行中的控制台输出信息写入.sql文件，即生成了备份文件。注：如果不对控制台信息进行读出，则会导致进程堵塞无法运行
			InputStream is = child.getInputStream();// 控制台的输出信息作为输入流
			InputStreamReader isReeder = new InputStreamReader(is, "utf8");// 设置输出流编码为utf8。这里必须是utf8，否则从流中读入的是乱码

			String inStr;
			StringBuffer sb = new StringBuffer("");
			// 组合控制台输出信息字符串
			BufferedReader br = new BufferedReader(isReeder);
			while ((inStr = br.readLine()) != null) {
				sb.append(inStr + "\r\n");
			}
			String outStr = sb.toString();

			// 要用来做导入用的sql目标文件：
//			File file = new File(filepath);
//			if(!file.exists()) file.createNewFile();
			FileOutputStream fous = new FileOutputStream(filepath);
			OutputStreamWriter writer = new OutputStreamWriter(fous, "utf8");
			writer.write(outStr);
			// 注：这里如果用缓冲方式写入文件的话，会导致中文乱码，用flush()方法则可以避免
			writer.flush();

			// 别忘记关闭输入输出流
			is.close();
			isReeder.close();
			br.close();
			writer.close();
			fous.close();

		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 * 导入数据库
	 */
	public static void restore(String username,String password,String database,String charset,String filepath) {
		try {
			Runtime rt = Runtime.getRuntime();
			// 调用 mysql 的 cmd:
			String cmd = "mysql -u"+username+" -p"+password+" "+database;
			Process process = rt.exec(cmd);
			OutputStream os = process.getOutputStream();// 控制台的输入信息作为输出流
			String inStr;
			List<String> lines=FileUtils.readLines(new File(filepath), "utf8");
			IOUtils.writeLines(lines,null, os, "utf8");

		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
	public static void main(String[] args){
		//backup("root","root","mcard","utf-8","c:/mcard1.sql");
		restore("root","root","mcard","utf-8", "c:/mcard090715.sql");
	}
}
